package com.snu.liwei.lyl_api.tools;

import org.springframework.util.StringUtils;
import org.springframework.web.multipart.MultipartFile;

import java.io.*;
import java.nio.charset.StandardCharsets;
import java.nio.file.Files;
import java.util.ArrayList;
import java.util.Base64;
import java.util.List;
import java.util.Objects;

public class FileUtils {

    //输出文件------------------------------------------------------------------------------------------------------------

    /**
     * 输出文件到硬盘
     *
     * @param path       输出地址
     * @param jsonObjectStr json字符串 对象
     * @throws IOException io异常
     */
    public static void outFile(String path, String jsonObjectStr) throws IOException {
        // 路径
        File file = new File(path);
        // 判断路径是否存在，不存在就创建
        if (!file.getParentFile().exists()) {
            boolean mkdir = file.getParentFile().mkdir();
            if (!mkdir) {
                throw new RuntimeException("创建文件夹异常");
            }
        }
        boolean newFile = file.createNewFile();
        if (!newFile) {
            //删除
            File tempt = new File(file.getPath());
            boolean delete = file.delete();
            if (delete) {
                file = tempt;
            }else {
                throw new RuntimeException("创建文件异常");
            }
        }
        // 写入的路径 和 编码格式
        Writer writer = new OutputStreamWriter(Files.newOutputStream(file.toPath()), StandardCharsets.UTF_8);
        // 将数据写入
        writer.write(jsonObjectStr + "\n");
        writer.flush();
        writer.close();
    }

    /**
     * 输出文件到硬盘
     *
     * @param path       输出地址
     * @param multipartFile 文件对象
     * @throws IOException io异常
     */
    public static void outFile(String path, MultipartFile multipartFile) throws IOException {
        // 路径
        File file = new File(path);
        // 判断路径是否存在，不存在就创建
        if (!file.getParentFile().exists()) {
            boolean mkdir = file.getParentFile().mkdir();
            if (!mkdir) {
                throw new RuntimeException("创建文件夹异常");
            }
        }
        multipartFile.transferTo(file);
    }

    //读取文件------------------------------------------------------------------------------------------------------------
    /**
     * 读取文件内容
     *
     * @param filePath String 如 c:\\1.txt 绝对路径
     * @return boolean
     */
    public static String readFile(String filePath) {
        StringBuilder fileContent = new StringBuilder();
        try {
            File f = new File(filePath);
            if (f.isFile() && f.exists()) {
                InputStreamReader read = new InputStreamReader(Files.newInputStream(f.toPath()), StandardCharsets.UTF_8);
                BufferedReader reader = new BufferedReader(read);
                String line;
                while ((line = reader.readLine()) != null) {
                    fileContent.append(line);
                }

                reader.close();
            }
        } catch (Exception e) {
            System.out.println("读取文件内容操作出错");
            e.printStackTrace();
        }
        return fileContent.toString();
    }


    //获取文件------------------------------------------------------------------------------------------------------------
    /**
     * 获取指定文件夹下所有文件，不含文件夹里的文件
     *
     * @param dirFilePath 文件夹路径
     * @return 文件夹下所有文件
     */
    public static List<File> getAllFile(String dirFilePath) {
        if (!StringUtils.hasText(dirFilePath))
            return null;
        return getAllFile(new File(dirFilePath));
    }


    /**
     * 获取指定文件夹下所有文件，不含文件夹里的文件
     *
     * @param dirFile 文件夹
     * @return 文件夹下所有文件
     */
    public static List<File> getAllFile(File dirFile) {
        // 如果文件夹不存在或着不是文件夹，则返回 null
        if (Objects.isNull(dirFile) || !dirFile.exists() || dirFile.isFile())
            return null;

        File[] childrenFiles = dirFile.listFiles();
        if (Objects.isNull(childrenFiles) || childrenFiles.length == 0)
            return null;

        List<File> files = new ArrayList<>();
        for (File childFile : childrenFiles) {
            // 如果是文件，直接添加到结果集合
            if (childFile.isFile()) {
                files.add(childFile);
            }
            // 以下几行代码取消注释后可以将所有子文件夹里的文件也获取到列表里
//            else {
//                // 如果是文件夹，则将其内部文件添加进结果集合
//                List<File> cFiles = getAllFile(childFile);
//                if (Objects.isNull(cFiles) || cFiles.isEmpty()) continue;
//                files.addAll(cFiles);
//            }
        }
        return files;
    }

    //BASE64解码成File文件
    public static void base64ToFile(String destPath,String base64, String fileName) {
        File file = null;
        //创建文件目录
        File  dir=new File(destPath);
        if (!dir.exists() && !dir.isDirectory()) {
            dir.mkdirs();
        }
        BufferedOutputStream bos = null;
        java.io.FileOutputStream fos = null;
        try {
            if (base64.contains("data:")) {
                int start = base64.indexOf(",");
                base64 = base64.substring(start + 1);
            }
            byte[] bytes = Base64.getDecoder().decode(base64);
            file=new File(destPath +"/"+fileName);
            fos = new java.io.FileOutputStream(file);
            bos = new BufferedOutputStream(fos);
            bos.write(bytes);
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            if (bos != null) {
                try {
                    bos.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
            if (fos != null) {
                try {
                    fos.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
    }

}
